import sys
from api import download_bilibili_video

def main():
    print("=" * 40)
    print("       B站视频及弹幕下载工具       ")
    print("=" * 40)
    print("请先在浏览器登录B站，获取SESSDATA和bili_jct并填入api.py中")
    print("说明：输入视频地址进行下载，输入'q'或'quit'退出程序\n")
    
    while True:
        video_url = input("请输入B站视频地址: ").strip()
        
        # 支持退出功能
        if video_url.lower() in ['q', 'quit']:
            print("感谢使用，程序即将退出...")
            break
            
        # 处理空输入
        if not video_url:
            print("⚠️ 视频地址不能为空，请重新输入\n")
            continue
            
        # 执行下载
        download_bilibili_video(video_url)
        print("\n------------------------------")
        print("准备处理下一个视频（输入'q'退出）\n")
    
    print("\n程序已退出")

if __name__ == "__main__":
    # 确保环境编码为UTF-8
    if sys.stdout.encoding != 'utf-8':
        sys.stdout.reconfigure(encoding='utf-8')
    main()
